package com.youth.oauth.commons.utils;


import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;

import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * JWT工具类
 */
public class JwtUtil {

    // 私有静态常量：密钥
    private static final String SECRET_KEY = "kcisxa2025!";
    // 私有静态常量：过期时间
    private static final long EXPIRATION_TIME = 60 * 60 * 1000; // 1小时

    /**
     * 生成JWT令牌
     *
     * @param username 用户名
     * @return 生成的JWT令牌
     */
    public static String generateToken(String username) {
        return Jwts.builder()
                .setSubject(username)
                .setExpiration(new Date(System.currentTimeMillis() + EXPIRATION_TIME))
                .signWith(SignatureAlgorithm.HS256, SECRET_KEY)
                .compact();
    }

    /**
     * 生成包含角色信息的JWT令牌
     *
     * @param username 用户名
     * @param roles 用户角色列表
     * @return 生成的JWT令牌
     */
    public static String generateToken(String username, List<String> roles) {
        Map<String, Object> claims = new HashMap<>();
        claims.put("roles", roles);
        return Jwts.builder()
                .setClaims(claims)
                .setSubject(username)
                .setExpiration(new Date(System.currentTimeMillis() + EXPIRATION_TIME))
                .signWith(SignatureAlgorithm.HS256, SECRET_KEY)
                .compact();
    }

    /**
     * 从令牌中提取用户名
     *
     * @param token JWT令牌
     * @return 提取的用户名
     */
    public static String getUsernameFromToken(String token) {
        return Jwts.parser()
                .setSigningKey(SECRET_KEY)
                .parseClaimsJws(token)
                .getBody()
                .getSubject();
    }

    /**
     * 从令牌中提取角色信息
     *
     * @param token JWT令牌
     * @return 提取的角色信息列表
     */
    public static List<String> getRolesFromToken(String token) {
        Claims claims = Jwts.parser().setSigningKey(SECRET_KEY)
                .parseClaimsJws(token).getBody();
        List<String> roles = (List<String>) claims.get("roles");
        return roles;
    }

    /**
     * 检查令牌是否过期
     *
     * @param token JWT令牌
     * @return 如果令牌已过期返回true，否则返回false
     */
    public static boolean isTokenExpired(String token) {
        Date expiration = Jwts.parser()
                .setSigningKey(SECRET_KEY)
                .parseClaimsJws(token)
                .getBody()
                .getExpiration();
        return expiration.before(new Date());
    }
}
